import org.junit.Test;
import org.omg.PortableServer.LIFESPAN_POLICY_ID;

import java.util.HashMap;
import java.util.Map;

public class Solution1171 {
    /**
     * 给你一个链表的头节点 head，请你编写代码，反复删去链表中由 总和 值为 0 的连续节点组成的序列，直到不存在这样的序列为止。
     * <p>
     * 删除完毕后，请你返回最终结果链表的头节点。
     * <p>
     * 你可以返回任何满足题目要求的答案。
     */
    public ListNode removeZeroSumSublists(ListNode head) {
        ListNode node = new ListNode(0);
        node.next = head;
        Map<Integer, ListNode> map = new HashMap<>();
        int sum = 0;
        map.put(0, node);
        while (head != null) {
            sum += head.val;
            map.put(sum, head);
            head = head.next;
        }

        sum = 0;
        head = node;
        while (head != null) {
            sum += head.val;
            head.next = map.get(sum).next;
            head = head.next;
        }
        return node.next;
    }

    @Test
    public void test(){
        ListNode node1 = new ListNode(1);
        ListNode node2 = new ListNode(2);
        ListNode node3 = new ListNode(3);
        ListNode node4 = new ListNode(-3);
        ListNode node5 = new ListNode(4);
        node1.next = node2;
        node2.next = node3;
        node3.next = node4;
        node4.next = node5;
        removeZeroSumSublists(node1);
    }
}

